#ifndef MYLOGGER_H
#define MYLOGGER_H

#include <iostream>
#include <string>

#include "NvInferRuntimeCommon.h"

class MyLogger:public nvinfer1::ILogger
{
public:
    MyLogger();

    // ILogger interface
public:
    void log(Severity severity, const char *msg);
    void log(const char *msg);
};

#endif // MYLOGGER_H
